package io.renren.common.dynamicdatasource;

/**
 * @author yoqu
 * @date 2017年07月25日
 * @time 下午1:55
 * @email wcjiang2@iflytek.com
 */
public class DataSourceHolder {
    public static final String DATABASE_BASE = "base";

    public static final String DATABASE_DSP = "dsp";

    //线程本地环境
    private static final ThreadLocal<String> dataSources = new ThreadLocal<String>();

    //设置数据源
    public static void setDataSource(String customerType) {
        dataSources.set(customerType);
    }
    //获取数据源
    public static String getDataSource() {
        return (String) dataSources.get();
    }
    //清除数据源
    public static void clearDataSource() {
        dataSources.remove();
    }
}
